package com.youlai.boot.system.service.impl;

import lombok.RequiredArgsConstructor;
import org.springframework.stereotype.Service;
import com.baomidou.mybatisplus.core.metadata.IPage;
import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
import com.youlai.boot.system.mapper.WasteTypeMapper;
import com.youlai.boot.system.service.WasteTypeService;
import com.youlai.boot.system.model.entity.WasteType;
import com.youlai.boot.system.model.form.WasteTypeForm;
import com.youlai.boot.system.model.query.WasteTypeQuery;
import com.youlai.boot.system.model.vo.WasteTypeVO;
import com.youlai.boot.system.converter.WasteTypeConverter;

import java.math.BigDecimal;
import java.util.Arrays;
import java.util.List;

import cn.hutool.core.lang.Assert;
import cn.hutool.core.util.StrUtil;

/**
 * 垃圾类型管理服务实现类
 *
 * @author youlaitech
 * @since 2025-02-22 16:33
 */
@Service
@RequiredArgsConstructor
public class WasteTypeServiceImpl extends ServiceImpl<WasteTypeMapper, WasteType> implements WasteTypeService {

    private final WasteTypeConverter wasteTypeConverter;
    private final WasteTypeMapper wasteTypeMapper;

    /**
     * 获取垃圾类型管理分页列表
     *
     * @param queryParams 查询参数
     * @return {@link IPage<WasteTypeVO>} 垃圾类型管理分页列表
     */
    @Override
    public IPage<WasteTypeVO> getWasteTypePage(WasteTypeQuery queryParams) {
        Page<WasteTypeVO> pageVO = this.baseMapper.getWasteTypePage(
                new Page<>(queryParams.getPageNum(), queryParams.getPageSize()),
                queryParams
        );
        return pageVO;
    }

    /**
     * 获取垃圾类型管理表单数据
     *
     * @param id 垃圾类型管理ID
     * @return
     */
    @Override
    public WasteTypeForm getWasteTypeFormData(Long id) {
        WasteType entity = this.getById(id);
        return wasteTypeConverter.toForm(entity);
    }

    /**
     * 新增垃圾类型管理
     *
     * @param formData 垃圾类型管理表单对象
     * @return
     */
    @Override
    public boolean saveWasteType(WasteTypeForm formData) {
        WasteType entity = wasteTypeConverter.toEntity(formData);
        return this.save(entity);
    }

    /**
     * 更新垃圾类型管理
     *
     * @param id   垃圾类型管理ID
     * @param formData 垃圾类型管理表单对象
     * @return
     */
    @Override
    public boolean updateWasteType(Long id, WasteTypeForm formData) {
        WasteType entity = wasteTypeConverter.toEntity(formData);
        return wasteTypeMapper.updateById(entity) > 0;
    }

    /**
     * 删除垃圾类型管理
     *
     * @param ids 垃圾类型管理ID，多个以英文逗号(,)分割
     * @return
     */
    @Override
    public boolean deleteWasteTypes(String ids) {
        Assert.isTrue(StrUtil.isNotBlank(ids), "删除的垃圾类型管理数据为空");
        // 逻辑删除
        List<Long> idList = Arrays.stream(ids.split(","))
                .map(Long::parseLong)
                .toList();
        return this.removeByIds(idList);
    }

    /**
     * 计算用户收益
     * 1. 盈利类（回收价>处理价）：
     *   用户收益 = 市场价*重量 + (回收价-处理价)*重量*用户分成比 + 碳补贴*重量
     * 2. 补贴类（回收价<=处理价）：
     *   用户收益 = 市场价*重量 + 政府补贴*重量*用户分成比 + 碳补贴*重量
     *
     */
    @Override
    public BigDecimal calculateUserRevenue(BigDecimal weight, WasteType wasteType) {
        BigDecimal marketPrice = wasteType.getMarketPrice();
        BigDecimal unitPrice = wasteType.getUnitPrice();
        BigDecimal processPrice = wasteType.getProcessPrice();
        BigDecimal cercSubsidyPrice = wasteType.getCercSubsidyPrice();
        BigDecimal govSubsidyPrice = wasteType.getGovSubsidyPrice();
        BigDecimal userRatio = wasteType.getUserRatio();

        if (unitPrice.compareTo(processPrice) > 0) {
            // 回收价 > 处理价 [盈利类]
            return marketPrice.multiply(weight)
                    .add(unitPrice.subtract(processPrice).multiply(weight).multiply(userRatio))
                    .add(cercSubsidyPrice.multiply(weight));
        } else {
            // 回收价 <= 处理价 [补贴类]
            return marketPrice.multiply(weight)
                    .add(govSubsidyPrice.multiply(weight).multiply(userRatio))
                    .add(cercSubsidyPrice.multiply(weight));
        }
    }
    /**
     * 计算机构收益
     * 1. 盈利类（回收价>处理价）：
     *   机构收益 = (回收价-处理价)*重量*机构分成比 + 环保评级奖励
     * 2. 补贴类（回收价<=处理价）：
     *   机构收益 = 政府补贴*重量*机构分成比 + (回收价+政府补贴-处理价)*重量 + 环保评级奖励
     * 注：环保评级奖励 => 基础环保奖励(50元/吨) × env_rating × 重量
     */
    @Override
    public BigDecimal calculateOrgRevenue(BigDecimal weight, WasteType wasteType) {
        BigDecimal unitPrice = wasteType.getUnitPrice();
        BigDecimal processPrice = wasteType.getProcessPrice();
        BigDecimal govSubsidyPrice = wasteType.getGovSubsidyPrice();
        BigDecimal orgRatio = wasteType.getOrgRatio();
        Integer envRating = wasteType.getEnvRating();

        if (unitPrice.compareTo(processPrice) > 0) {
            // 回收价 > 处理价 [盈利类]
            return unitPrice.subtract(processPrice).multiply(weight).multiply(orgRatio)
                    .add(new BigDecimal("50").multiply(new BigDecimal(envRating)).multiply(weight));
        }else{
            // 回收价 <= 处理价 [补贴类]
            return govSubsidyPrice.multiply(weight).multiply(orgRatio)
                    .add(unitPrice.add(govSubsidyPrice).subtract(processPrice).multiply(weight))
                    .add(new BigDecimal("50").multiply(new BigDecimal(envRating)).multiply(weight));
        }
    }

}
